#!/bin/bash

# 设置颜色
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo -e "${GREEN}开始部署人脸搜索服务...${NC}"

# 检查 Python 版本
python3 --version || {
    echo -e "${YELLOW}请先安装 Python 3.x${NC}"
    exit 1
}

# 创建虚拟环境
echo -e "${GREEN}创建虚拟环境...${NC}"
python3 -m venv .venv

# 激活虚拟环境
echo -e "${GREEN}激活虚拟环境...${NC}"
source .venv/bin/activate

# 安装依赖
echo -e "${GREEN}安装依赖...${NC}"
pip install -r requirements.txt

# 检查并创建必要的目录
echo -e "${GREEN}创建必要的目录...${NC}"
mkdir -p logs
mkdir -p temp_uploads
mkdir -p uploads

# 设置环境变量
export APP_ENV=production
export PYTHONPATH=$PYTHONPATH:$(pwd)

# 启动服务
echo -e "${GREEN}启动人脸搜索服务...${NC}"
nohup python face_search_service.py > logs/app.log 2>&1 &

# 保存进程ID
echo $! > .pid

echo -e "${GREEN}服务已启动，进程ID: $!${NC}"
echo -e "${GREEN}查看日志: tail -f logs/app.log${NC}"